package com.heima.tencentcloud.scan;

import com.heima.common.constants.cloud.tencent.TencentConstants;
import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.ims.v20201229.ImsClient;
import com.tencentcloudapi.ims.v20201229.models.ImageModerationRequest;
import com.tencentcloudapi.ims.v20201229.models.ImageModerationResponse;
import com.tencentcloudapi.tms.v20201229.TmsClient;
import com.tencentcloudapi.tms.v20201229.models.TextModerationRequest;
import com.tencentcloudapi.tms.v20201229.models.TextModerationResponse;
import lombok.Getter;
import lombok.Setter;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.PropertySource;
import org.springframework.stereotype.Component;

import java.util.Base64;
import java.util.Collection;

/**
 * @Author XHui
 * @Since 2024/2/12 12:54
 * @Version 1.0
 */
@Getter
@Setter
@Component
@PropertySource("classpath:tencentcloud.properties")
@ConfigurationProperties(prefix = "tencentcloud")
public class ContentSecurityScan {

    private String accessKeyId;
    private String secret;
    private String scenes;

    // 扫描图片通过腾讯云的SDK
    public ImageModerationResponse imageScan(String imageUrl) throws TencentCloudSDKException {

            // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
            // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
            // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
            Credential cred = getCredential();

            // 实例化一个http选项，可选的，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            httpProfile.setEndpoint("ims.ap-beijing.tencentcloudapi.com");

            // 实例化一个client选项，可选的，没有特殊需求可以跳过
            ClientProfile clientProfile = new ClientProfile();
            clientProfile.setHttpProfile(httpProfile);

            // 实例化要请求产品的client对象,clientProfile是可选的
            ImsClient client = new ImsClient(cred, "ap-beijing", clientProfile);

            // 实例化一个请求对象,每个接口都会对应一个request对象
            ImageModerationRequest req = new ImageModerationRequest();
            req.setFileUrl(imageUrl);

            // 返回的resp是一个ImageModerationResponse的实例，与请求对象对应
            ImageModerationResponse resp = client.ImageModeration(req);
            // 输出json格式的字符串回包
            //FormatterUtil.formatPrintAsJson(resp);
            return resp;
    }

    public ImageModerationResponse imageScanBatch(Collection<String> imageUrls) throws TencentCloudSDKException {
        // 1.初始化map
        if(imageUrls == null || imageUrls.size() == 0){
            return null;
        }
        ImageModerationResponse currentResp = null;
        for (String imageUrl: imageUrls) { // 2.循环调用接口，
            currentResp = imageScan(imageUrl);
            if(TencentConstants.CONTENT_SUGGESTION_BLOCK
                    .equals(currentResp.getSuggestion())){
                return currentResp; // 出现违规图像，直接返回false
            }else if(TencentConstants.CONTENT_SUGGESTION_REVIEW
                    .equals(currentResp.getSuggestion())){
                return currentResp; // 出现疑似违规图像，直接返回null, 转为人工审核
            }
        }
        return currentResp; // 都是正常图像，返回true
    }

    public TextModerationResponse textScan(String content) throws TencentCloudSDKException {
            // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
            // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
            // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
            // 1.
        Credential cred = getCredential();

        // 实例化一个http选项，可选的，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            httpProfile.setEndpoint("tms.tencentcloudapi.com");

            // 实例化一个client选项，可选的，没有特殊需求可以跳过
            ClientProfile clientProfile = new ClientProfile();
            clientProfile.setHttpProfile(httpProfile);

            // 实例化要请求产品的client对象,clientProfile是可选的
            TmsClient client = new TmsClient(cred, "ap-beijing", clientProfile);

            // 实例化一个请求对象,每个接口都会对应一个request对象
            TextModerationRequest req = new TextModerationRequest();
            String base64Content = Base64.getEncoder().encodeToString(content.getBytes());
            req.setContent(base64Content);

            // 返回的resp是一个TextModerationResponse的实例，与请求对象对应
            TextModerationResponse resp = client.TextModeration(req);
            // 输出json格式的字符串回包
            //FormatterUtil.formatPrintAsJson(resp);
            // System.out.println(resp);
            return resp;

    }

    private Credential getCredential() {
        return new Credential(
                accessKeyId,
                secret);
    }

}
